/*
 * Copyright 2020-2030 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.gitee.minimalismstyle.fresh.common.data.jdbc.beetl

import org.beetl.sql.core.SQLManager
import org.beetl.sql.ext.spring.SpringConnectionSource
import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty
import org.springframework.context.ApplicationContext
import org.springframework.context.ApplicationContextAware
import org.springframework.context.annotation.Configuration
import javax.annotation.PostConstruct
import javax.annotation.Resource
import javax.sql.DataSource

/**
 * BeetlSqlDataSource主从配置
 * @author maoxiaodong
 */
//@Configuration
//@ConditionalOnProperty(value = ["beetlsql.mutiple.datasource"], havingValue = "master,slave")
class FreshBeetlSqlDataSourceConfig : ApplicationContextAware {

    private var applicationContext: ApplicationContext? = null


    @PostConstruct
    fun setBeetlSqlDataSource() {
        this.applicationContext!!.getBeansOfType(SQLManager::class.java).forEach { (_, sqlManager) ->
            val beetlSqlDataSource = SpringConnectionSource()
            beetlSqlDataSource.masterSource = master
            beetlSqlDataSource.slaveSource = arrayOf(slave)
            sqlManager.ds = beetlSqlDataSource
        }
    }

    @Resource(name = "master")
    lateinit var master: DataSource

    @Resource(name = "slave")
    lateinit var slave: DataSource


    override fun setApplicationContext(applicationContext: ApplicationContext) {
        this.applicationContext = applicationContext
    }

}